from rest_framework.generics import ListAPIView
from rest_framework.viewsets import ModelViewSet

from apps.meiduo_admin.serializers.user import UserModelSerializer
from apps.meiduo_admin.utils import PageNum
from apps.users.models import Users


class UserModelViewSet(ModelViewSet):
    # 指定查询集
    # queryset = Users.objects.all()

    # 指定序列化器
    serializer_class = UserModelSerializer

    # 指定分页
    pagination_class = PageNum

    # 不指定指定查询集时调用的方法
    # 重写查询方法 简单实现过滤
    def get_queryset(self):
        keyword = self.request.query_params.get("keyword")

        if keyword:
            return Users.objects.filter(username__contains=keyword)
        return Users.objects.all()
